输入门(Input Gate)
概述 (50-200字符)
LSTM 三个门之一,决定将什么新信息写入细胞状态。由 iₜ(控制更新量,∈[0,1])和 C̃ₜ(候选新信息,∈[-1,1])组成,与遗忘门协作完成细胞状态的精确更新。
关键内容 (≥300字符, 用双链)
- 数学公式:iₜ = σ(Wi · [hₜ₋₁, xₜ] + bi),控制更新量 [0, 1];C̃ₜ = tanh(Wc · [hₜ₋₁, xₜ] + bc),候选新信息 [-1, 1]。两个子组件分别使用 Sigmoid激活函数 和 tanh 激活函数。
- 功能直觉:输入门决定哪些新信息应该被写入 细胞状态(Cell State)。在语言模型中,遇到"小明"时输入门会写入主语信息;遇到"他"时保留"他=小明"的关联。iₜ 接近 1 时大量写入新信息,接近 0 时阻止写入。
- 细胞状态更新:输入门与 遗忘门(Forget Gate) 协作:Cₜ = fₜ ⊙ Cₜ₋₁ + iₜ ⊙ C̃ₜ。遗忘门控制"擦除旧信息",输入门控制"写入新信息",两者共同实现细胞状态的精确调控。
- 候选值 C̃ₜ:使用 tanh 而非 sigmoid 生成候选值,输出范围 [-1, 1],允许细胞状态存储正负两种方向的信息,增强了表达能力。
来源
- Long Short-Term Memory (1997 论文) — Hochreiter & Schmidhuber (1997) 提出输入门概念
- raw/articles/ai-papers/machine-learning/04_lstm_1997.md — 输入门的数学公式与语言学直觉
相关
- LSTM(长短期记忆网络) — part_of(LSTM 的核心组件)
- 细胞状态(Cell State) — relates_to(控制细胞状态的更新)
- 遗忘门(Forget Gate) — relates_to(协作更新细胞状态)
- Sigmoid激活函数 — uses(iₜ 的激活函数)
- 输出门(Output Gate) — relates_to(LSTM 三門之一)